package config

import (
	"fmt"
	"github.com/joho/godotenv"
	"os"
	"strconv"
)

func init() {
	//加载.env文件
	err := godotenv.Load()
	if err != nil {
		fmt.Println("没有找到.env文件", err)
		os.Exit(1) //让当前程序以给出的状态码 code 退出
	}
}

// GetEnv 获取对应配置
func GetEnv(key string) string {
	return os.Getenv(key)
}

// GetEnvInt 获取对应配置并转int
func GetEnvInt(key string) int {
	val := os.Getenv(key)
	ret, err := strconv.Atoi(val)
	if err != nil {
		return 0
	} else {
		return ret
	}
}

func GetEnvBool(key string) bool {
	val := os.Getenv(key)
	return val == "true"
}
